package com.lb.pdm;

import com.baomidou.mybatisplus.annotation.FieldFill;
import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.generator.FastAutoGenerator;
import com.baomidou.mybatisplus.generator.config.OutputFile;
import com.baomidou.mybatisplus.generator.config.rules.DateType;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;
import com.baomidou.mybatisplus.generator.fill.Column;

import java.util.Collections;

/**
 * MybatisPlus代码生成器
 * 
 * @author pdm
 * @since 2024-01-15
 */
public class CodeGenerator {

    /**
     * 数据库连接配置
     */
    private static final String URL = "jdbc:mysql://8.140.225.7:4406/pdm?useUnicode=true&characterEncoding=UTF-8&autoReconnect=true&useSSL=false&allowPublicKeyRetrieval=true&serverTimezone=UTC";
    private static final String USERNAME = "root";
    private static final String PASSWORD = "root";


    private static final String PROJECT_PATH = System.getProperty("user.dir");
    private static final String JAVA_PATH = PROJECT_PATH + "/src/main/java";
    private static final String MAPPER_XML_PATH = PROJECT_PATH + "/src/main/resources/mapper";

    /**
     * 包名配置
     */
    private static final String PARENT_PACKAGE = "com.lb.pdm";
    private static final String ENTITY_PACKAGE = "entity";
    private static final String MAPPER_PACKAGE = "mapper";
    private static final String SERVICE_PACKAGE = "service";
    private static final String SERVICE_IMPL_PACKAGE = "service.impl";
    private static final String CONTROLLER_PACKAGE = "controller";

    /**
     * 作者信息
     */
    private static final String AUTHOR = "pdm";

    public static void main(String[] args) {
        // 需要生成的表名，可以修改为实际的表名
        String[] tableNames = {
            "sys_user",
            "person_info", 
            "household_info",
            "household_member",
            "birth_registration",
            "household_transfer",
            "family_reunion",
            "floating_population",
            "key_personnel",
            "certificate_info",
            "certificate_application",
            "sys_dict"
        };

        // 执行代码生成
        generateCode(tableNames);
    }

    /**
     * 执行代码生成
     * 
     * @param tableNames 表名数组
     */
    private static void generateCode(String[] tableNames) {
        FastAutoGenerator.create(URL, USERNAME, PASSWORD)
                // 全局配置
                .globalConfig(builder -> {
                    builder.author(AUTHOR) // 设置作者
                            .enableSwagger() // 开启 swagger 模式
                            // .fileOverride() // 已移除：该方法在新版本MyBatis-Plus中已废弃，无需显式覆盖
                            .outputDir(JAVA_PATH) // 指定输出目录
                            .dateType(DateType.ONLY_DATE) // 时间类型对应策略
                            .commentDate("yyyy-MM-dd"); // 注释日期格式
                })
                // 包配置
                .packageConfig(builder -> {
                    builder.parent(PARENT_PACKAGE) // 设置父包名
                            .entity(ENTITY_PACKAGE) // 设置实体类包名
                            .mapper(MAPPER_PACKAGE) // 设置 Mapper 接口包名
                            .service(SERVICE_PACKAGE) // 设置 Service 接口包名
                            .serviceImpl(SERVICE_IMPL_PACKAGE) // 设置 Service 实现类包名
                            .controller(CONTROLLER_PACKAGE) // 设置 Controller 包名
                            .pathInfo(Collections.singletonMap(OutputFile.xml, MAPPER_XML_PATH)); // 设置mapperXml生成路径
                })
                // 策略配置
                .strategyConfig(builder -> {
                    builder.addInclude(tableNames) // 设置需要生成的表名
                            .addTablePrefix("t_", "c_") // 设置过滤表前缀
                            // 实体类策略配置
                            .entityBuilder()
                            .enableLombok() // 开启 lombok 模型
                            .enableTableFieldAnnotation() // 开启生成实体时生成字段注解
                            .naming(NamingStrategy.underline_to_camel) // 数据库表映射到实体的命名策略
                            .columnNaming(NamingStrategy.underline_to_camel) // 数据库表字段映射到实体的命名策略
                            .idType(IdType.AUTO) // 主键类型
                            .addTableFills(new Column("create_time", FieldFill.INSERT)) // 自动填充配置
                            .addTableFills(new Column("update_time", FieldFill.INSERT_UPDATE))
                            // Mapper策略配置
                            .mapperBuilder()
                            .enableMapperAnnotation() // 开启 @Mapper 注解
                            .enableBaseResultMap() // 启用 BaseResultMap 生成
                            .enableBaseColumnList() // 启用 BaseColumnList
                            // Service策略配置
                            .serviceBuilder()
                            .formatServiceFileName("%sService") // 格式化 service 接口文件名称
                            .formatServiceImplFileName("%sServiceImpl") // 格式化 service 实现类文件名称
                            // Controller策略配置
                            .controllerBuilder()
                            .enableRestStyle() // 开启生成@RestController 控制器
                            .enableHyphenStyle(); // 开启驼峰转连字符
                })
                // 模板引擎配置，使用 Freemarker 引擎模板
                .templateEngine(new FreemarkerTemplateEngine())
                .execute();

        System.out.println("代码生成完成！");
        System.out.println("生成路径：" + JAVA_PATH);
        System.out.println("Mapper XML路径：" + MAPPER_XML_PATH);
    }
}